#include <stdio.h>
#include "ch11.h"

// 传入一个字符串，以及长度
char * s_gets(char * st, int n)
{
    char * ret_val;
    int i = 0;

    ret_val = fgets(st, n, stdin);      // fgets() 返回指向char的指针，正常情况下返回的地址和第一个参数相同
    if (ret_val) // 即 ret_val != NULL
    {
        while (st[i] != '\n' && st[i] != '\0')
            i++;
        if (st[i] == '\n')
            st[i] = '\0';
        else
            while (getchar() != '\n')
                continue;            
    }
    
    return ret_val;
}